data_dir = '/home/indico/datasets/iclr2016'